package cc.shacocloud.luanniao.core.model.po

import cc.shacocloud.luanniao.core.model.constant.DatasetsStatusEnum
import cc.shacocloud.luanniao.core.model.constant.DatasetsTypeEnum
import cc.shacocloud.luanniao.core.model.constant.TrainingMethodEnum
import org.springframework.data.relational.core.mapping.Column
import org.springframework.data.relational.core.mapping.Table

/**
 * 数据集
 * @author 思追(shaco)
 */
@Table("def_datasets")
data class DatasetsPo(

    /**
     * 关联知识库 id
     */
    @Column("knowledge_base_id")
    val knowledgeBaseId: Long,

    /**
     * 名称
     */
    @Column("name")
    var name: String,

    /**
     * 训练方式
     */
    @Column("training_method")
    val trainingMethod: TrainingMethodEnum,

    /**
     * 字数
     */
    @Column("word_count")
    val wordCount: Int,

    /**
     * 类型
     */
    @Column("type")
    var type: DatasetsTypeEnum,

    /**
     * 数据集状态
     */
    @Column("status")
    var status: DatasetsStatusEnum,
) : BasePo()